from ultralytics import YOLO
import os

# 加载预训练模型
model = YOLO('yolov8s.pt')

# 开始训练
model.train(
    data='dataset.yaml',  # 数据集配置文件
    epochs=100,           # 训练轮数
    imgsz=640,            # 图像尺寸
    batch=16,             # 批次大小
    # device='0'
)
